# 使用tushare包获取某股票的历史行情数据
# 使用pandas包计算该股票历史数据的5日均线和30日均线
# 使用matplotlib包可视化历史数据的收盘价和两条均线
# 分析输出所有金叉日期和死叉日期
# 如果我从假如我从2010年1月1日开始，初始资金为100000元，金叉尽
# 量买入，死叉全部卖出，则到今天为止，我的炒股收益率如何?
import numpy as np
import pandas as pd
import matplotlib.pyplot as plt
import tushare as ts
#
df=pd.read_csv("601318.csv",index_col='date',parse_dates=['date'])[['open','close','high','low']]
df['ma5']=np.nan
df['ma30']=np.nan
#
# for i in range(4,len(df)):
#     df.loc[df.index[i],'ma5']=df['close'][i-4:i+1].mean()
# for i in range(29,len(df)):
#     df.loc[df.index[i],'ma30']=df['close'][i-29:i+1].mean()
df['ma5']=df['close'].rolling(5).mean()
df['ma30']=df['close'].rolling(30).mean()
#df=df[:1000]
df[['close','ma5','ma30']].plot()
plt.show()
